import torch
import torch.nn as nn
import torch.optim as optim
import data_loader
import utils
from actuator import Trainer

# # 加载数据
# trainloader = data_loader.load_mnist_data()
#
# # 定义模型
# model = model.SimpleCNN()
#
# # 定义损失函数和优化器
# criterion = nn.CrossEntropyLoss()
# optimizer = optim.Adam(model.parameters(), lr=0.001)
#
# # 训练模型
# train.training_model(model, trainloader, criterion, optimizer, num_epochs=5)


def training_model():

    trainer = Trainer(
        output_dir='datas/output'
    )

    trainer.training(
        n_epochs=1,
        checkpoint_path=None,
        lr=0.01,
        weight_decay=0.0
    )


if __name__ == '__main__':
    training_model()

